from django.test import TestCase

from .mymath import add


class MymathTest(TestCase):

    def test_add_1_2(self):
        r = add(1, 2)
        self.assertEqual(r, 3)

    def test_add_a_b(self):
        r = add('a', 'b')
        self.assertEqual(r, 'ab')
